package com.leetcode.p100;

public class Test198 {

//    public int rob(int[] nums) {
//        if (nums.length == 0) {
//            return nums[0];
//        }
//
//        if (nums.length == 1) {
//            return nums[0];
//        }
//
//        if (nums.length == 2) {
//            return Math.max(nums[0], nums[1]);
//        }
//
//        int[] dp = new int[nums.length];
//        dp[0] = nums[0];
//        dp[1] = Math.max(nums[0], nums[1]);
//
//        for (int i = 2; i < nums.length; i++) {
//            dp[i] = Math.max(nums[i] + dp[i - 2], dp[i - 1]);
//        }
//
//        return dp[nums.length - 1];
//    }

    /**
     * 采用 滚动数组 机制，节省内存，其实上面的更通俗易懂
     */
    public int rob(int[] nums) {
        if (nums.length == 0) {
            return nums[0];
        }

        if (nums.length == 1) {
            return nums[0];
        }

        if (nums.length == 2) {
            return Math.max(nums[0], nums[1]);
        }

        int pre = nums[0];
        int next = Math.max(nums[0], nums[1]);
        int result = 0;

        for (int i = 2; i < nums.length; i++) {
            result = Math.max(nums[i] + pre, next);
            pre = next;
            next = result;
        }

        return next;
    }

    public static void main(String[] args) {
        Test198 test = new Test198();
        int result = test.rob(new int[]{1, 2, 3, 1});
        System.out.println(result);
    }

}
